基于层级树索引器的次二次方注意力机制
🧭 层级树索引稀疏注意力加速长上下文
Zyphra Research
提出层级树结构索引器对KV cache递归压缩,使注意力总成本随序列长度亚线性增长,实现次二次方缩放
📌 工作参考价值(4 点)
- 电商长序列建模:用户历史行为、商品序列、会话记录可视为 KV cache,借鉴层级树压缩摘要,在推理时自上而下检索相关 token,实现长上下文稀疏注意力,降低 inference cost。
- 轻量适配存量基座模型:仅需数亿 token 的 KL 蒸馏,业务侧可为已有 LLM 加装该索引器,无需重新预训练。
- 树深度稳健:即使 6 层深树也不显著掉点,工程上可自由调整索引层级,在精度与延迟间取平衡。
- Agent 场景收益:长上下文 agentic coding 性能不降,适合需要超长工具调用历史 / 文档上下文的电商导购 Agent 或智能客服,可直接扩展 context window 而不更换模型。
GistRank:基于 Gist 压缩的大模型列表级排序
🧭 LLM listwise 重排 · Gist 压缩 ★ 深度解读
Southwest University · Renmin University of China · Yuanbao Team, Tencent · Institute of Computing Technology, Chinese Academy of Sciences
用三阶段训练将每个候选段落压成4个gist token,在TREC/BEIR上以约14倍加速逼近全文本重排效果
📌 工作参考价值(4 点)
- 把商品标题/详情/卖点用 Stage 1 的 query 预测目标预训练成 4 个 gist token,线上重排只拼接 gist,可将 100 个候选的 prompt 从 8k+ token 降到约 1.25k,延迟和 KV cache 占用大幅下降
- 借鉴 Stage 2 的 citation/click 偏好:在电商搜索中可用点击 vs 曝光、成交 vs 点击构造成对数据,训练 gist 或 item 表示具备判别性,无需昂贵人工标注;再用少量人工/LLM 标注做 Stage 3 listwise 微调
- 用 rank-slot + ListMLE 非自回归打分替代 LLM 自回归生成排序,可以把排序阶段从生成式 token 序列变成一个固定大小的打分矩阵,吞吐更稳定,适合在线 serving
- 注意成本结构:候选压缩占 90%+ 延迟,优先考虑对全量商品库离线预计算 query-independent gist 或做近似缓存,只在线对当前 query 做轻量融合/重排序
PFA:长序列生成式推荐中注意力可渐进聚焦
🧭 生成式推荐 · 稀疏注意力加速 ★ 深度解读
Alibaba International Digital Commerce Group · Zhejiang University
利用 GR 解码中注意力渐进聚焦与跨步排名一致性,PFA 通过共享重要性排名和几何预算缩减,加速长序列生成式推荐
📌 工作参考价值(4 点)
- 对长序列生成式推荐,可借鉴"早期构造共享重要性排名 + 后续几何缩减预算",替代每步每 beam 独立 top-k,消除重复选择和 KV gather 开销,尤其适合 beam width 大、序列长的电商召回/广告候选生成。
- 工程实现上,用 beam 生成概率加权聚合注意力分布来挑选关键历史 token,比每个 beam 独立选更稳定且能共享连续 KV cache;可先 reorder KV cache,后续直接前缀截断,降低非连续内存访问。
- Step-aware KV projection:为不同解码步分配独立 KV 投影参数,成本低但能显著缓解复用 token 带来的表示退化,适合层级 SID 生成。
- 若业务采用生成式推荐,可保留 Step 0 全注意力做全局探索,不用过早压缩历史,对长序列效果稳定。
TAO:面向LLM Agent服务的工具感知联合调度框架
🧭 Agent 服务调度优化
Shanghai Jiao Tong University · Stanford University
提出工具感知联合调度框架TAO,融合工具就绪度、缓存信用与token占用进行动态规划调度,显著降低延迟和TTFT
📌 工作参考价值(4 点)
- 在 LLM Agent 服务中,工具调用(如商品检索、数据库查询、API 调用)会使 GPU 推理可中断,调度时需将「工具返回就绪时间」纳入准入/放置决策,避免 GPU 空转和资源浪费。
- 借鉴 TAO 的 value interface:组合「近期工具就绪度 + 节点级缓存信用(时间衰减)+ token footprint」来评估一个 session 的调度价值,而不是只看队列长度或显存。
- 动态规划联合选择等待中的 sessions 并放置到 GPU 节点,在容量约束下最大化整体收益,比启发式 admission/placement 更优;可实现在线调度器,在 GPU 空闲或工具返回事件时触发。
- 如果推荐/搜索 Agent 涉及多步工具调用(如商品召回、属性查询、库存检查),可以对工具返回时间做预测或标记,并结合 prefix cache 命中(例如共享 system prompt 或工具 schema)优先调度相似 session 以提高 decode throughput。
TokenGate:可学习 token 剪枝加速 LLM 预填充且精度不降
🧭 LLM 长上下文预填充加速 · 可学习 token 剪枝 ★ 深度解读
Stanford University · UC Berkeley
提出可插入组件 TokenGate,通过类 MoE 的 token-gated attention 端到端训练 token 剪枝,大幅加速长上下文预填充且精度不降
📌 工作参考价值(4 点)
- 在电商/Agent 场景中,长文档(商品详情、用户评论、搜索广告上下文、RAG 检索文档)的预填充是延迟和内存瓶颈;可借鉴 TokenGate 在模型中间层引入可学习 token 剪枝,相比 FastKV 等启发式方法,端到端训练能显著提升速度且精度无损,且可与 LoRA 一起微调,训练开销仅约 1.3x/step,总体甚至更少。
- token-gated attention 是关键 trick:把不可微的 top-K 剪枝通过类似 MoE 的门控值乘到 attention logits 上,让 scoring network 获得梯度,同时让模型骨干自适应剪枝后的稀疏注意力;该思路可迁移到任何需要在 Transformer 中间层做软筛选的任务,如多文档 RAG 中对检索片段做软裁剪。
- 训练时使用 dual budget sampling(全量 + 对数均匀采样的剪枝预算各一半损失),使单个模型泛化到不同 token 预算,无需为每个压缩比重新训练;这对线上需要动态调整 TTFT 和内存的推荐 / Agent 系统非常实用,可以一次训练,推理时按需调整 K。
- 工程实现上,将 per-key gate bias 融合进 FlashAttention 内核、剪枝后 forward 用 CUDA graph 固定 padding、KV cache 选择性裁剪等做法可直接复用到自研推理引擎,降低集成成本。
DLite:面向 Agentic AI 的无历史投机解码框架
🧭 LLM 推理加速 · 历史无关投机解码
Shanghai Jiao Tong University · Shanghai Innovation Institute · Tsinghua University · Infinigence-AI
DLite 复用目标模型隐藏状态进行局部投机预测,避免维护历史 KV cache,在 agentic 任务中接受长度达 2-6 并超 DSpark 最高 70%
📌 工作参考价值(3 点)
- 在电商/推荐系统的多轮对话 agent(如客服、导购)中,长历史上下文会让 draft model 接受长度骤降且 KV cache 开销大;可借鉴 DLite 的 history-free 思路:不单独维护 draft model 的历史 KV cache,直接复用 target model 的隐藏状态做局部预测,大幅降低显存和延迟。
- 两阶段训练策略(先对齐隐藏状态再训练 sequential head 恢复块内因果依赖)适合训练自定义轻量 draft 模块,在效果和资源间取得平衡,可复用于需要低延迟的在线推理服务。
- 局部简单预测替代全局上下文建模,在 agentic 多轮交互中接受长度仍能保持 2-6,说明牺牲部分长程依赖换取更低的系统开销是可行的工程折中,适合高并发场景。
TransX 技术报告:面向推荐系统的序列转导规模化
🧭 序列转导推荐 · 模型-基础设施协同 ★ 深度解读
LinkedIn
提出 TransX 序列转导框架,将行为编码、记忆与候选融合解耦,通过摊销执行实现模型扩容而不等比增加在线成本
📌 工作参考价值(4 点)
- 商品/广告精排或召回中,候选集常达数百到上千,可借鉴 TransX 的共享行为记忆 + 候选条件融合:把用户长期行为压缩成 global-local memory,在线用 GQA 交叉注意力对候选做融合,避免每个候选都重算历史表示,显著降低在线 GPU 成本。
- 服务架构上,把用户行为编码从在线路径剥离,改为离线 bulk prefill + 近线增量 refresh,在线只消费不可变行为 memory KV 快照。适合电商推荐中用户行为更新频率远低于请求频率的场景,但需注意版本一致性、KV 量化与分层存储。
- 训练侧可复用多前缀共享一次编码 + blocked prefix scan + batched local gather:在长序列行为模型训练时,消除多个 event cutoff 的重复前缀编码,实测吞吐提升 84%,MFU 从 41% 提升到 57%,显存峰值下降约 44%,对大用户行为序列训练很有价值。
- 通过共享行为记忆接口进行多任务后训练(检索、排序、生成式推荐),避免为每个任务重新训练 backbone,适合业务中同时存在召回、精排和生成式推荐需求的团队降低维护成本。
自适应 DHiSparse:负载自适应分层 KV 上的投机解码
🧭 长上下文 LLM 推理 · 分层 KV 投机 ★ 深度解读
Umans AI
把 DSpark/DFlash2 投机解码接入分层稀疏 KV,并用可回收 optional residency 随负载调整 GPU 驻留,显著提升长上下文吞吐
📌 工作参考价值(4 点)
- 长上下文 Agent / LLM4Rec 场景:不要全量 resident 或全量 offload,用「固定 hot cache + 可回收 optional history」;低负载多驻留 GPU,高负载按整请求回收,host 永远有备份,能同时保吞吐和并发。
- 接入投机解码时,draft KV 用窗口 ring 限制大小,按 W + Kmax 加 page 对齐分配;验证阶段多个 query 的 top-k 选择做 union 去重,同一 miss row 只 host copy 一次,本文中可减少约 29% fetch。
- verification staging 按 live layer group 复用,而不是给每层分配;GLM 78 个 payload 层只留 4 个 arena,staging 从 24.0 GiB 降到 1.23 GiB,这对多 query 验证和长上下文服务非常关键。
- 调度上可用式(13)判断投机是否划算:当 speculative 与 non-speculative 的 active session 比低于 TPOT 改善比时,投机不再增加吞吐,应动态收窄 block 或关闭投机,避免高负载下白白等待 slot。
Rollout 引导的 MoE LLM 强化学习 FP4 量化感知训练框架
🧭 低精度 rollout 加速 MoE LLM 强化学习训练
Alibaba Token Hub, Alibaba Group · Ohio State University
提出 TRACE,通过 rollout 侧量化结果引导训练侧 FP4 舍入,实现 FP4 rollout 与 BF16 性能相当并最高加速 5.4 倍
📌 工作参考价值(4 点)
- 在 RL 调 LLM 推荐/搜索 Agent 或生成式 ranker 时,rollout 阶段是主要算力瓶颈,可尝试 FP4 权重/激活 + FP4 KV cache 联合做 rollout;TRACE 表明可与 BF16 rollout 获得相近的最终策略质量。
- 避免直接用离线 PTQ 量化 BF16 训练好的策略:把 rollout 侧实际量化噪声(舍入结果、scale)作为监督信号引入训练侧 QAT,能显著改善最终 FP4 策略性能,比事后量化更可靠。
- 工程上可借鉴 selective quantization-info caching:只缓存深层 mantissa/scale,降低多 GPU 或分布式 rollout worker 间的存储与通信开销,对 MoE 或超大 LLM 生成服务有直接参考。
- 低精度 rollout 最高 5.4 倍加速意味着相同算力下可大幅增加采样轨迹数量,适合需要大量探索的 query 推荐、对话式推荐 Agent 等 RL 训练场景。
基于选择的结构化推理:迈向高效多模态搜索智能体
🧭 Agent推理效率优化·选择式结构化推理 ★ 深度解读
Apple · Carnegie Mellon University
将多模态搜索Agent的逐步推理从自由生成改为从候选库中并行选择,保持成功率且推理延迟降超90%
📌 工作参考价值(4 点)
- 在电商导购/搜索 Agent 中,把每轮规划从自由生成改成路由选择:预先定义 5-10 条高频决策候选,如“澄清需求 / 生成搜索 query / 比价 / 读图识别 / 基于证据作答”,让模型按上下文似然打分选择,再用选中候选引导 action 解码。这样能保留自然语言 guidance 的效果,同时规避自回归推理的延迟不可控。
- 工程上可直接复用“并行 prefill 打分”的做法:对候选文本做 teacher-forced logprob 计算,共享历史 KV cache,不需要额外分类头或 SFT warm-up;用 SGLang 的 logprob_start_len + max_new_tokens=0 可实现。候选库可离线维护,p95 延迟比自由生成稳定得多,适合在线流量。
- 训练层面,若用 RL 训练 Agent,可把“规划/路由选择”当作 categorical action,动作 token 保持常规 importance ratio;用 GRPO 更稳,GSPO/SAPO 多步更新时 selector-gradient 近似会损失精度,业务上可采用更多 rollout 或一次更新。
- 候选设计要注意两点:不要只打分 index,要打分完整自然语言文本的语义;库规模从 1 到 6 带来看得见的提升,所以可以先用日志聚类/高绩效轨迹总结出高频意图,再逐步扩展到 6-10 条。
HELIX:净化并统一大规模推荐中的特征交互与序列建模
🧭 统一特征交互与序列建模的排序架构 ★ 深度解读
Data - Global E-Commerce Recommendation Video Team
用单向可复用 KV cache 与 Mixup-PerToken-FFN 联合扩展特征交互和序列建模,电商视频 GMV/用户提升约 6%
📌 工作参考价值(4 点)
- 把候选无关的用户行为序列编码成可复用 KV cache,候选侧只做 cross-attention 读取,强制单向信息流。这样在 serving 多候选、training 多样本时都能摊销 encoder 成本;如果搜广推系统里要接入用户长序列,这是可落地的架构选择。
- 异构特征交互可考虑用 Mixup-PerToken-FFN 替代 self-attention:参数免费的 Mixup 按通道切片重组 token,再用 token-wise SwiGLU 做非线性交互,比 self-attention 更适配稠密异构特征,且省计算。
- 序列 encoder 采用金字塔 query 压缩:按特征重要性给每条序列分配最近 token 预算,而不是全局截断或统一池化;配合 padding-free 变长执行,能显著降低长序列训练成本。
- 训练稳定性上,对 SwiGLU 的 gate/up 分支做非参数归一化可以避免 FP16 激活溢出;QK-Norm 和 value-side gated attention 在推荐排序模型里也有稳定 AUC 增益。
SparseEngine:稀疏优先的LLM推理引擎
🧭 稀疏注意力 · KV cache推理引擎 ★ 深度解读
Harbin Institute of Technology
通过统一生命周期契约支持15种稀疏注意力方法,并提供Chain Cache与可控前缀缓存剪枝,实现长上下文agent推理10倍吞吐提升
📌 工作参考价值(4 点)
- 生命周期 hook 抽象可借鉴到自研推理服务:将稀疏方法集成点解耦为 SparseController(注册回调)、CacheManager(方法特定状态存储)和 AttentionView(连接注意力后端),可以不改模型代码快速接入多种稀疏/压缩策略,适合电商 Agent 服务中灵活切换 KV 优化方案。
- Chain Cache 对多轮 Agent 场景非常实用:跨请求保留被 eviction 方法裁剪过的 KV 及元数据(如 H2O 累积分数),能避免每轮重新 prefill 完整历史,对电商客服、推荐 agent 等多轮交互可显著降低首字延迟和重复计算。
- 可控 prefix-cache pruning 允许业务方按价值管理上下文:应用可以指定历史区间(如低价值的工具输出)和保留比例,用稀疏策略(如 KVzip)选择保留位置,同时保持逻辑前缀匹配,适合在广告/推荐 agent 中清理过期或冗余的会话内容而不破坏缓存复用。
- 实验结论表明物理 KV 驱逐是吞吐提升的关键:在相同 GPU 内存下,SnapKV 物理驱逐可支撑约10倍更大 batch,带来约10倍 aggregate decode throughput;动态稀疏注意力在匹配并发下可获1.5-2.6倍 decode 吞吐提升,且质量几乎无损(LongBench 平均差异仅 +0.17 分),可作为电商长上下文 LLM 推理优化的直接依据。
WUSH-KV:数据自适应变换的 KV Cache 低比特量化
🧭 LLM 推理优化 · KV Cache 量化 ★ 深度解读
Institute of Science and Technology Austria (ISTA) · ETH Zürich · Red Hat AI
用 WUSH 数据自适应可逆变换对 KV cache 做低比特量化,2-bit 下 Qwen3-8B 全面优于 OSCAR
📌 工作参考价值(4 点)
- 对需要长上下文/高并发 LLM 服务的推荐与 Agent 系统,KV cache 内存是主要瓶颈。可借鉴将可逆变换与量化解耦:校准阶段按每个 head 的 Gram 矩阵(缓存张量自身统计)+ Hessian(下游消费者敏感度)构建变换,允许非正交各向异性缩放,比固定 Hadamard/随机旋转在 2-bit 下显著降低误差。类似思路可用于用户/物品 embedding 缓存或特征存储的低比特压缩,关键是 Hessian 选下游任务对扰动的敏感度。
- 值侧变换可在离线折叠进权重(V→WV、WO),在线零额外开销;键侧变换因 RoPE 无法折叠,放在 RoPE 之后,每新增 token 只多做一次 d×d 矩阵乘。工程上优先折叠 value 侧,key 侧用固定在线 transform,避免逐位置补偿。
- 保留 sink token 和最近 token 的 FP16 窗口(例如 sink=64, recent=256)并用批量量化 flush(chunk 8/16)可在大幅降低平均位宽时保护长上下文和生成质量;该策略与具体量化器无关,可直接迁移。
- 校准只需 128 条序列、一次完成,跨 bitwidth 复用;量化器可选 QuEST 或 OSCAR-style percentile-clipped affine,证明 WUSH 与 QuEST 搭配接近最优。在 SGLang 中集成时,可沿用 OSCAR 的 cache 管理策略,不必为 WUSH 重新调参。
Draft-KV:学习语言模型间有价值的潜在通信
🧭 LLM 潜在通信 · KV Cache 传输
City University of Hong Kong · University of Science and Technology of China · University of Electronic Science and Technology of China · AIPD, Tencent · Theory Lab, Huawei
提出 Draft-KV,用 sharer 草稿过程的 KV 状态经门控侧记忆接口传给 frozen 小模型,显著提升协同推理
📌 工作参考价值(4 点)
- 在多模型 Agent 链路中,让大模型(sharer)提前对候选 query/商品/用户意图草拟推理,并把草稿过程的 KV cache 暂存为侧记忆;线上由小模型接收这些 KV 做最终决策或生成,接口参数仅 ~1M,两端模型冻结,部署成本远低于全量微调。
- 接口设计可借鉴「线性投影 + 门控 attention 分支」:不把通信状态注入主隐层,而是作为 side memory 参与注意力,避免干扰小模型已有表示;门控可控制通信信息的权重,便于线上 A/B 开关和降级。
- 训练采用两阶段 progressive:先做 message reconstruction 对齐表示,再转到 answer supervision;并加入 mismatched message 的 guard 损失,防止通信缺失或错配时模型被带偏,这对推荐中的多路召回/多模型融合也有参考。
- 验证通信是否真被使用:用随机替换消息做控制实验,若替换后指标几乎不变,说明增益来自接口结构而非信息内容;在推荐/广告引入 LLM 通信或增强时,应做同样消融,避免伪提升。
激活记忆与参数记忆在小样本学习中的互补作用
🧭 LLM记忆机制与测试时训练
Northeastern University, Shenyang, China · NiuTrans Research, Shenyang, China
系统比较LLM中激活记忆与参数记忆的差异,发现事实回忆靠激活记忆,复合任务需二者协同
📌 工作参考价值(4 点)
- 在电商/Agent 的上下文管理中,事实性信息(如商品属性、历史订单)放进 KV cache 或 prompt 即可稳定回忆,不必为此做在线参数更新。
- 对需要“先查事实再执行新规则”的复合任务(如促销规则叠加用户偏好),仅靠 ICL 或仅靠微调都不够,架构上应同时保留激活记忆和轻量参数更新两条路径。
- 神经元分析表明两类记忆激活不同神经元群:模型可解释性排查时,可判断失败原因是上下文丢失还是参数适应不足,再针对性修复。
- 整体为机制研究,业务可借鉴点有限,但可作为混合记忆设计(RAG + 测试时训练)的理论依据。
Flash-dLLM:IO感知KV缓存与并行解码加速扩散LLM
🧭 扩散LLM推理加速·KV cache
VILA Lab, MBZUAI
训练无关Flash-dLLM以IO感知融合KV cache和自草稿-验证解码,显著加速扩散LLM推理并省内存
📌 工作参考价值(4 点)
- 在 LLM 推理服务(如广告文案/搜索词生成)中,KV cache 的 HBM 读写常成为瓶颈;可借鉴 Flash-dLLM 的 IO-aware 融合 kernel,把多个注意力/cache 操作合并执行,减少冗余显存搬运,提升 batch/long-context 吞吐。
- 自草稿-验证(self draft-and-verify)不引入额外 drafter 模型,适合资源受限的 Agent/推荐系统:让主模型复用 KV cache 同时生成候选并按验证结果接受,降低延迟且不增加显存。
- 其并行解码与 KV-cache 驱动候选验证思路可迁移到批量生成场景,例如离线批量生成商品标题、query 改写或 push 文案;考虑非自回归或半自回归解码来扩大 batch size、缩短首 token 延迟。
- 面对 RAG 长文档或用户长序列,KV cache 显存与 I/O 受限时,可参考其对缓存复用和显存效率的优化策略,设计更紧凑的 cache 分层或融合模式。
工具调用进度自报:Agent 服务系统应读取而非猜测
🧭 Agent 工具调用 · KV cache 调度
Tsinghua University · Alibaba Cloud Computing · Zhejiang University
让工具调用上报剩余进度,比预估算更准,可将 p90 TTFT 降低约 20%
📌 工作参考价值(4 点)
- 在电商/搜索 Agent 中,对长耗时工具(SQL 查询、库存 API、商品爬虫、订单处理)增加 progress 上报字段,如剩余比例或“即将完成”信号,服务层可用它做 KV cache 驱逐/恢复决策,而不是靠工具名或历史均值猜时长。
- 工程实现上,progress 信号通过 harness 旁路采集,不改变 Agent 可见的输入输出,对 benchmark 分数无损耗;这可以直接迁移到现有多 Agent 编排框架,无需修改模型 prompt。
- 在推理引擎侧,把工具 progress 作为轻量 hint 接入 KV cache 管理,作者在 LRU 上仅做小改动就接近 oracle,说明该信号比复杂预估器更有投产价值;如果业务受 TTFT p90 波动困扰,可优先采工具真实进度而非加预测模型。
- 进度信号在环境变化时仍然准确,适合电商大促、突发流量等场景;对多租户 GPU 资源调度,长工具调用按真实剩余时间释放/保留缓存,能提升 HBM 利用率。
文档应该存在哪里:上下文、表示还是参数?
🧭 知识注入位置对比 · KV vs LoRA ★ 深度解读
Amazon AGI
系统对比知识注入三种位置,Cartridges(KV)在多文档检索中唯一匹配 ICL,但高压缩下有遗忘
📌 工作参考价值(4 点)
- 电商/搜索的高频文档(商品详情、活动规则、广告素材)可以预计算成 KV-cache adapter(Cartridge)在线复用,避免每次查询重复 prefill 原文;20× 压缩下加载 10 个 adapter 约 30–50ms,10× token 减少约带来 100× prefill FLOPs 下降,适合低延迟在线服务。
- 多文档组合场景下,不要用「每个文档训一个 LoRA 再合并」替代检索,尤其信息密集任务;论文显示一个 distractor 就能让合并效果崩溃(FinQA k=3 时 34.5→4.9)。若必须用参数化方式,joint training 优于 adapter merging,但仍不如可拼接的 KV-cache。
- 训练知识注入 adapter 时,用蒸馏目标(teacher = 原文 in context,student = adapter)比普通 next-token CE 平均高 4 个点,在数值推理/金融类任务上收益更大;可直接迁移到商品属性提取、活动规则问答等场景。
- 上线前要评估通用能力退化:LoRA 低秩约束更稳,full FT 和大 MLP adapter 遗忘严重;如果使用 Cartridge,可考虑用 Compaction 初始化 KV cache 来缓解 HumanEval 类代码能力下降(论文中下降了 16 点)。
面向智能手术室的语音交互多智能体系统:架构设计与关键技术
🧭 多智能体语音交互 · KV Cache 优化
Wuhan United Imaging Surgical Co., Ltd. (UIS)
用 LLM 多智能体实现手术室语音控制与报告生成,重点优化 KV Cache 前缀复用、流式解析并行执行与渐进式提示披露
📌 工作参考价值(4 点)
- KV Cache prefix warming 可直接用于 Agent 长 system prompt 或设备状态频繁变化场景:将稳定的前缀(角色定义、技能列表、历史上下文)缓存,每次请求只重算变化部分。对电商导购 Agent,用户切换商品详情或筛选条件时,不必重算整个对话前缀,可把首字延迟从几百毫秒降到几十毫秒。
- 流式 partial JSON 解析 + 提前并行执行是降低端到端时延的有效工程 trick:LLM 以流式输出任务数组时,不用等完整 JSON 结束,解析到单个完整任务对象就立即调度执行。对广告文案批量生成、多商品并行推荐解释、多工具调用编排,能带来约 30% 的时延下降。
- 渐进式技能提示披露适合工具多的 Agent 系统:按用户角色、可用设备/工具、当前阶段动态裁剪 system prompt,只保留当前上下文最相关的技能说明。电商场景可复用为:按用户等级、可用的推荐/营销工具、会话阶段动态注入工具 schema,避免上下文浪费,提升长会话下的指令跟随稳定性。
- 多智能体架构与设备管理层解耦,可参考其 skill registry、task planner、device manager 的分层设计,在推荐系统中把“用户意图理解”“策略编排”“召回/排序执行”拆成独立模块,便于管控和低延迟并行执行。
记忆即计划:基于记忆接地规划的世界-动作建模
🧭 Agent 长程记忆与计划执行解耦
MaP-WAM 将长期记忆转化为稀疏视觉上下文与语言计划,由固定上下文执行器遵循,取得 RMBench 83.3% 成功率
📌 工作参考价值(4 点)
- 在需要长用户历史的推荐/Agent 场景,可借鉴“记忆即计划”:把长时间跨度的行为序列压缩为若干关键帧(或代表性 item 特征)+ 高层意图,供 planner 生成下一步计划,而非让 executor 每次都读全量历史。
- 计划与执行分离:规划器可以处理长历史并生成轻量计划,执行器只接收固定长度上下文和当前计划,结合 progress 预估实现自适应切换,在线推理延迟与历史长度解耦;结构化注意力支持 KV cache 可进一步降低重复规划成本。
- 对电商视觉/短视频推荐,如果任务依赖细粒度视觉证据,语言摘要会丢失信息,可保留用户最近强交互的视觉特征帧作为稀疏长期记忆,而不是仅存文本标签。
- 但这是机器人操作领域,所提 WAP 和进度对齐机制较复杂;直接迁移到推荐/广告需要简化:progress 可类比用户状态/会话阶段,可改造为 session 切分与 memory refresh 信号。
ShallowStream:浅层索引、深层回答的流式视频理解框架
🧭 多模态流式视频高效推理
Harbin Institute of Technology (Shenzhen)
用 MLLM 浅层同时做帧编码与索引构建,查询时按浅层注意力打分并多样性选帧,延迟降低最高 52.1 倍且性能对齐强基线
📌 工作参考价值(4 点)
- 直播电商/短视频理解场景可用“浅层预填充建索引、深层仅对选中片段推理”的两段式架构:常驻只保留浅层 KV cache,在线视频帧不断写入轻量索引,查询时只对高相关帧做完整深度计算,大幅降低延迟与显存。
- 用浅层 attention score 做 query-aware 帧打分,不需要额外训练检索器;类似推荐里用注意力分数筛选候选 item 或行为序列,能把用户当前 query / 意图与视觉片段相关性评估集中到轻量模型。
- diversity-aware selection 可直接迁移到推荐/广告视频素材或直播切片选取:在相关帧集合上做多样性采样,避免连续重复/相似画面霸屏,提升信息覆盖与用户体验。
- 工程上可借鉴“按 prefill depth 分层 KV cache”思路:在线流式处理时低频深算、高频浅算,对长视频/直播流式多模态 LLM 服务有直接成本收益。
语言模型可以控制自身注意力
🧭 LLM 长上下文稀疏注意力 ★ 深度解读
KAIST AI · Google DeepMind
提出声明式注意力协议,让模型在思维链中显式声明注意力范围,零训练实现长上下文稀疏注意力
📌 工作参考价值(4 点)
- 在长上下文多轮对话(电商客服、购物助手、推荐解释 Agent)中,把历史对话、商品信息、检索结果切分为可寻址片段,让模型通过 <focus> 标签声明需要关注的片段,可减少 50% 以上的 KV cache 读取,适合大 batch 在线服务降延迟。
- 与 RAG/工具调用结合:将检索到的商品/广告/知识片段作为模拟工具返回的“magic chunk”,模型在推理中自行选择关注哪些片段,避免每步全量扫描,可显著降低多文档推理成本。
- 工程上可直接复用论文的 vLLM 集成思路:通过 hook 改写 KV cache block table 实现块对齐掩码,不改内核、兼容 FlashAttention,适合快速在现有推荐/搜索 Agent 服务中落地。
- DA 的注意力计划以文本形式输出,可审计模型在推荐解释、搜索排序中具体关注了哪些输入,用于优化上下文构造和检索策略;未来还可结合 KV cache offloading 实现可逆上下文压缩,适合长会话 Agent。
HyQuant:面向 LLM 注意力的混合精度量化框架
🧭 LLM 长上下文推理 · 混合精度量化 ★ 深度解读
Shanghai Jiao Tong University · Xi’an Jiaotong University · Tencent Penglai Lab · Xiamen University
保留 vertical-line token 与局部窗口全精度、其余 KV 低比特,实现长上下文推理的近无损量化加速
📌 工作参考价值(4 点)
- 长上下文/Agent 场景做 KV cache 量化时,不要全局等精度:按列注意力质量选择 top-5% vertical-line token + 最近 W=128 窗口保留 FP16/BF16,其余 4-bit,可在 LongBench/GSM8K 上接近全精度;该模式可直接迁移到电商导购 Agent 的长期对话历史、RAG 长 prompt 缓存。
- 工程实现上,把 dequantize 融合进 attention kernel 的 online softmax 扫描,避免先材料化全精度 KV;在 H100 上 32K 前缀 decode kernel 相对 FA2 达 3.58×、端到端 1.17×;对广告/搜索的 LLM serving 能降低显存带宽压力,显存不足时也可以跑更大 batch。
- Vertical-line 识别只需累加 attention 概率列和,额外 runtime 约 3-5%;用 tail-query proxy 且每 64 tokens 更新,避免逐 token 全局打分,可作为轻量重要性评估器与稀疏/eviction 策略配合。
- 局限:短上下文收益有限,且未在 coding/agentic 任务评估;接入长提示或 long CoT 前应先在自家任务验证 vertical-line 稳定性。
FlashPrefill V2:面向长上下文LLM服务的块稀疏预填充注意力
🧭 长上下文推理·块稀疏注意力优化
MAIS & NLPR, CASIA · UCAS · WeChat, Tencent
通过均值修正、FA3/4对齐内核与分页KV/连续批处理,将块稀疏预填充从原型推进到生产级,128K下最高加速47倍
📌 工作参考价值(4 点)
- 用户长序列(如点击序列、会话历史)的预填充延迟是长上下文推荐/Agent系统的首要瓶颈,可借鉴块稀疏注意力:用均值修正项在极端稀疏下稳定精度,避免稀疏化后排序/检索效果明显下降。
- PackGQA内存访问、warp specialization和pingpong pipelining的组合可直接指导自研稀疏attention kernel,尤其适合在H20等推理卡上部署FP8量化,显著降低长上下文服务成本。
- 原生支持paged KV cache与continuous batching,说明稀疏注意力可作为现代推理框架(如SGLang)的backend无缝接入,适合在电商/搜索推荐在线服务中做透明替换,无需改动上层模型逻辑。
- 如果业务中有大规模多轮对话或长文档预填充,可参考块稀疏阈值机制做动态跳块,并在FP8精度下验证效果,兼顾首token延迟与模型质量。
阶段回放分歧跟随KV缓存:固定前缀精度控制与双向缓存移植
🧭 LLM推理诊断·KV缓存与数值精度
Openhagen
发现推理阶段回放的分歧由KV缓存状态决定,FP32可消除BF16下的解码分歧,双向缓存移植证实因果性
📌 工作参考价值(4 点)
- 在需要严格复现生成结果的场景(如Agent多步推理、离线评估、A/B测试)中,必须保持KV缓存位的精确一致,仅依赖相同token ID和贪婪解码还不够;若使用BF16混合精度,意外分歧可能被误判为模型行为变化。
- 推荐使用FP32进行关键路径的推理或缓存存储,以避免因精度导致的隐性分歧;若必须使用BF16,需通过逐令牌桥接验证缓存等价性。
- 双向缓存移植技术可用于诊断生产环境中不可复现的生成轨迹,能快速定位分歧源头是模型权重还是缓存状态。
- 在构建基于LLM的推荐Agent时,若采用中间状态暂停-继续的策略,应固定前缀缓存并控制精度,避免因缓存重建引入不可控的推荐结果漂移。
ReToken:用一个Token提升视觉语言模型的长视觉检索
🧭 多模态视觉检索 · 稀疏Token选择
University of Illinois at Urbana-Champaign · Microsoft Research · Google DeepMind
用单个可学习Token从视觉KV缓存中检索稀疏相关token,大幅提升长视觉上下文中的问答准确率
📌 工作参考价值(4 点)
- 电商场景中处理商品多图或视频描述时,可利用可学习检索Token从大量视觉特征中快速定位与用户查询相关的帧/区域,提升问答或推荐理由生成的质量。
- 利用预填充的KV缓存和稀疏选择机制,避免每次查询都重新处理所有视觉token,大幅降低多模态推荐的推理延迟和显存占用,适合线上服务。
- 轻量级设计(单个token,单卡训练)表明,小型Query-Image数据集即可训练出有效的检索能力,可迁移到商品视觉问答或视觉搜索中。
- 该方法本质是条件特征选择,可扩展到用户行为序列(如视频浏览记录)中,用学习到的Token从长序列中检索相关历史行为,增强推荐模型而无需输入完整序列。
C²KV:压缩与可组合的KV缓存复用,实现高效长上下文LLM推理
🧭 LLM推理加速 · KV缓存复用与压缩
Shanghai Jiao Tong University · Alibaba Group
联合优化KV压缩与拼接,实现位置无关的可组合缓存,长上下文推理加速达17倍
📌 工作参考价值(4 点)
- 在构建基于RAG的推荐或多文档搜索系统时,可直接复用C²KV的压缩与可组合缓存,避免每次请求重新编码海量上下文,大幅降低首token延迟和GPU内存占用。
- 轻量侧车Extractor不修改基座模型,适合在已有生产LLM(如阿里通义、ChatGLM)上快速部署,无需重新训练或对齐,工程风险低。
- 位置无关的模块化缓存设计,允许不同来源的文档片段独立压缩后动态拼接,特别适合电商搜索中多商品描述、多用户评价等异构上下文组合场景。
- 压缩-拼接联合训练策略确保压缩后的表示可直接用于下游注意力计算,避免简单量化和重用带来的精度崩塌,为业务中长文本处理提供了可靠的效率-质量平衡。
KVpop:预测式在线剪枝的键值缓存压缩
🧭 LLM 推理 KV 缓存压缩与剪枝
NXAI · Johannes Kepler University Linz
通过监督未来注意力信号学习固定预算的 KV 缓存淘汰策略,结合延迟评分器,在推理任务上以高压缩比保持性能
📌 工作参考价值(4 点)
- 在电商搜索/推荐系统的 LLM 推理中,KV 缓存是长上下文(如用户行为序列、多轮对话)的瓶颈,KVpop 的固定预算淘汰策略可直接应用于减少显存占用,支持更长上下文而不崩溃。
- 延迟记忆评分器思路可借鉴:对用户行为序列建模时,不急于丢弃近期 token,等积累更多上下文后再判重要性,更准确地保留关键信息(如近期点击可能关联后续购买意图)。
- 未来注意力监督的训练方式可迁移到“查询预判”场景:训练一个小型打分器预测 token 对未来查询的效用,用于动态压缩 RAG 召回文档或对话历史。
- 工程部署中,KVpop 固定预算保证了内存使用上界,适合线上服务稳定运行,可集成到 vLLM 等推理框架中,降低大规模部署成本。
面向内存高效自回归视频生成的实例特定参数吸收方法
🧭 自回归生成 · 参数化KV缓存压缩
Institute of Information Engineering, Chinese Academy of Sciences · The University of Hong Kong · AMAP, Alibaba Group
ISPA首次将KV缓存压缩从丢弃转变为实例级参数蒸馏,在视频生成中实现50%缓存缩减且质量无损
📌 工作参考价值(4 点)
- 在推荐系统的长序列Transformer或生成式推荐模型中,可借鉴ISPA将部分层转为局部注意力,并通过最小二乘将历史缓存吸收到权重中,节省推理内存。
- 预热阶段监测全注意力和局部注意力的输出差异来确定转换点,可用于Agent对话历史压缩,动态决定何时将旧上下文编码进模型参数。
- 闭式解的权重调制无需重新训练,适合在线推理时实时进行缓存压缩,可直接集成到现有KVCache管理的工程框架中。
- 对于电商场景的实时生成(如商品描述、直播推流),用ISPA减少视频/文本自回归生成的KV缓存,提升吞吐。
ReFreeKV:迈向无阈值 KV 缓存压缩
🧭 LLM 推理优化 · KV缓存压缩
Nanjing University of Aeronautics and Astronautics · WeChat AI, Tencent · Fudan University · Independent Researcher
提出首个无阈值 KV 缓存压缩方法,自适应分配预算,无需预设阈值即可保持全缓存性能
📌 工作参考价值(4 点)
- 长上下文推理优化:在电商、广告场景中,用户行为序列、商品描述等长文本输入可利用该自适应 KV 缓存压缩,降低线上推理内存与延迟,无需为不同输入单独调阈值。
- 多轮 Agent 对话管理:Agent 系统常需维持长对话历史,ReFreeKV 的无阈值特性可动态保留关键 KV 对,避免对话状态因缓存裁剪过度而丢失重要信息,提升多轮交互稳定性。
- 避免输入敏感调参:业务中处理多样化的查询(搜索词、聊天消息)时,传统缓存压缩需凭经验设定预算,此方法可自动调整,减少工程维护成本。
- 模型部署降本:在 LLM 推荐、检索增强生成等场景下,直接复用该方法实现近乎无性能损失的 KV 缓存压缩,节省 GPU 显存,支持更大 batch size 或更长上下文,直接降低服务成本。
面向 Token 操作的大模型推理优化技术
🧭 大模型推理成本与效率优化
China Unicom (Data Intelligence / Data Science and AI Research Institute) · Tsinghua University · Peking University · Fudan University · Zhejiang Lab
首次提出四层 token 推理优化架构,系统梳理降本增效的产业技术路径
📌 工作参考价值(4 点)
- 模型路由与多模型融合:推荐系统常需多模型(精排、粗排、RAG等)协同,可借鉴论文的模型路由与融合调度策略,减少总推理调用量,降低 token 成本。
- KV Cache 命中与跨阶段复用:对话式推荐、Agent 记忆、长会话推荐中,跨请求的 KV Cache 缓存与预填充策略可直接降低首 token 延迟,提升交互体验。
- 量化与投机解码落地:对部署在 CPU/端侧的 LLM4Rec 模型,采用 INT4/INT8 量化、投机解码等典型技术可将推理成本压缩 3-5 倍,适合高并发推荐场景。
- 动态批处理与负载均衡:在推荐模型在线服务中,动态 batching 与跨节点的负载均衡可平滑流量峰值,保障推荐链路稳定性,尤其适合大促峰值保障。
可控稀疏的混合注意力:通过可学习分配优化FA/SWA组合
🧭 高效LLM推理 · 混合注意力稀疏分配 ★ 深度解读
Institute of Information Engineering, Chinese Academy of Sciences · School of Cyber Security, University of Chinese Academy of Sciences · Baidu Inc.
用L0正则化与增广拉格朗日约束学习全局/局部注意力的最优分配,实现可控稀疏并揭示底层局部、中层全局的一致结构
📌 工作参考价值(4 点)
- 部署长上下文LLM(如query理解、会话摘要)时,可用ConSA动态决定哪些层/头用全局注意力、哪些用滑动窗口,在几乎不降性能的同时减少KV cache和推理计算。
- 借鉴可控稀疏思想,在推荐系统的Transformer中为不同用户行为序列(短期 vs. 长期)分配不同的注意力范围,用类似L0-拉格朗日方法学习分配。
- 增广拉格朗日约束可精确满足目标稀疏率,适合在业务中需要严格控制资源预算的场景(例如单机GPU限制)。
- 手工设计混合注意力时可参考已发现的规律:底层放局部注意力、中层放全局注意力,避免均匀交错模式。
可变宽度 Transformer
🧭 Transformer 架构优化 · 宽度缩放
MIT · MIT-IBM Watson AI Lab
提出 × 形瓶颈架构,中间层窄、两端宽,同等参数量下性能更优且推理成本更低
📌 工作参考价值(4 点)
- 将瓶颈结构引入大模型,可减少推荐/搜索系统中 LLM 的显存和计算开销,尤其对需保留 KV cache 的在线服务有利
- 无需额外参数的残差维度映射方法可直接复用到现有 Transformer 推荐模型(如 SASRec),通过非均匀宽度提升效率
- 对于 Agent 中多次调用 LLM 的场景,可降低单次推理延迟与成本,加速多步决策
- 提示我们在设计生成式推荐模型时,不必每层等宽,中间层适当缩减可兼顾容量与效率
CONF-KV:面向长序列LLM的置信度感知KV缓存淘汰与混合精度存储
🧭 长上下文LLM推理 · 置信度驱动KV缓存淘汰
Carnegie Mellon University
利用模型生成时的置信度动态调整KV缓存预算,结合混合精度存储和分层金字塔预算,在低内存下保持长文本理解精度。
📌 工作参考价值(4 点)
- 电商Agent等长对话/工具调用场景中,KV缓存迅速膨胀,可借鉴置信度感知淘汰策略:当模型对下一token预测不确定时保留更多历史,减少关键信息丢失;预测置信时激进剪枝,节省显存。
- 混合精度存储(FP16/INT8)可有效压缩KV缓存,结合淘汰策略进一步降低内存占用,适合资源受限的在线服务部署。
- 金字塔逐层预算分配(浅层保留较多token,深层保留较少)可平衡信息保留与计算开销,可结合具体模型结构调整各层预算比例。
- 保护最近窗口(如固定最近若干token)保证局部连贯性,与置信度驱动预算结合,简单有效,可直接应用于现有Agent系统。
LCGuard:多智能体系统中安全 KV 共享的隐通信守卫
🧭 多智能体安全隐通信优化
Rensselaer Polytechnic Institute · IBM Research
通过对抗训练学习 KV 缓存变换,在保持任务性能的同时降低多智能体隐通信中的敏感信息泄露
📌 工作参考价值(4 点)
- 在多智能体联合推理的电商或客服场景中,若不同 Agent 通过 KV 缓存共享中间推理状态,可借鉴 LCGuard 的对抗训练框架,在缓存传输前插入可学习的表示变换层,降低用户隐私泄露风险。
- 可以将敏感信息重构成功率作为一种可量化的隐私泄露指标,用于评估和监控分布式 Agent 系统内部的通信安全性。
- 对抗训练的分治策略(保留任务语义、压制敏感信息)可推广到其他需要交换隐向量的推荐或检索组件中,例如生成式推荐中的 Semantic ID 传递。
- 工程实现上,LCGuard 的变换模块轻量、可插拔,不改变原有 Agent 结构和推理流程,适合已有流水线的逐步升级。
OScaR:极端KV缓存量化的奥卡姆剃刀
🧭 LLM推理优化 · INT2 KV缓存量化
Tsinghua University · Meituan LongCat Team · The University of Hong Kong · The University of Edinburgh · UCAS
OScaR通过Canalized Rotation与Omni-Token Scaling解决Token Norm Imbalance,实现INT2近乎无损KV缓存量化,解码加速3.0×
📌 工作参考价值(4 点)
- INT2 KV缓存部署:电商对话或实时推荐场景中,用OScaR的INT2量化可降低96%的KV缓存内存(5.3×),结合3.0×解码加速,适合高并发低延迟需求。
- 轻量集成路径:Canalized Rotation + Omni-Token Scaling无需复杂重采样或辅助模型,仅需少量显式变换,可直接插入现有Transformer推理管线,降低工程复杂度。
- 多模态量化统一:OScaR已覆盖文本、多模态及全模态LLM,对购物图文推荐中的多模态模型KV缓存压缩同样有效,避免为不同模态设计多套量化方案。
- CUDA Kernel优化:论文提供的定制化CUDA kernel将旋转与缩放无缝融合,可借鉴其极简高效的设计思路加速自研模型推理。
全注意力反击:百步训练内将全注意力转化为稀疏注意力
🧭 长上下文推理 · 注意力稀疏化
Nanjing University · Alibaba Group
发现全注意力LLM存在内在稀疏性,通过轻量索引和动态选择仅需数百步训练即可达到近乎无损的稀疏推理
📌 工作参考价值(4 点)
- 动态 top-p 筛选: 对电商搜索/推荐中的长序列用户行为编码,借鉴 query-dependent 的动态 token 预算,代替固定 top-k,可保留关键信息同时降低计算开销。
- 检索头识别与保留: 在 Agent 长对话记忆或商品详细页理解中,仅保留少数注意力头进行全长度上下文检索,其余头使用稀疏模式,可大幅压缩 KV 缓存。
- 低维索引器设计: 用 16 维索引矩阵替代完整注意力分数计算,适合实时性要求高的生成式检索或召回阶段,可在预填充阶段大幅提速。
- 少步迁移训练: 利用已有的全注意力模型,仅需数百步微调即可迁移到稀疏推理,无需从头预训练稀疏模型,节省训练资源且保持精度,适合业务快速迭代。
Forcing-KV: 面向高效自回归视频扩散模型的混合 KV 缓存压缩
🧭 Multimodal ★ 深度解读
ZJU · Video Rebirth · HKUST · BJTU
发现自回归视频扩散模型注意力头分为静态和动态两类,提出混合 KV 缓存压缩,结合静态结构剪枝与动态相似性剪枝,实现最高 2.82× 加速和 30% 显存减少